

package com.hazelcast.scheduledexecutor.impl;

import java.util.concurrent.RejectedExecutionException;

public class PartitionCapacityPermit implements CapacityPermit {

    private final String name;
    private final int partition;
    private final int capacity;
    private int permits;

    PartitionCapacityPermit(String name, int permits, int partition) {
        this.name = name;
        this.capacity = permits;
        this.permits = permits;
        this.partition = partition;
    }

    @Override
    public void acquire() throws RejectedExecutionException {
        if (permits <= 0) {
            throw new RejectedExecutionException("Maximum capacity (" + capacity + ") of tasks reached for partition (" + partition + ") " + "and scheduled executor (" + name + "). " + "Reminder, that tasks must be disposed if not needed.");
        }

        --permits;
    }

    @Override
    public void acquireQuietly() {
        --permits;
    }

    @Override
    public void release() {
        ++permits;
    }

    @Override
    public int totalAcquired() {
        return capacity - permits;
    }
}
